#!/usr/bin/python3
# -*- coding:utf-8 -*-
""""
   @Author: 叶建伟
   @Email: yejw46507@hundsun.com
   @Time: 2022/6/7 17:07
   @File: 专户产品管理_删除.py
"""
import time

from 基础公共脚本.步骤检查_hui2 import check_table, check_table_rows,FE
from 基础公共脚本.公共方法_hui2 import open_menu
from 基础公共脚本.common_util import click_delete,click_cancel,click_confirm

def special_product_delete(input_data=None, ui_expect=None):
    # 进入 风险等级查询 菜单 线下客户管理/线下客户管理
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '线下客户管理/专户产品管理'})
    # 资源设置搜索框表单填入
    time.sleep(2)

    # 资源设置搜索框表单填入
    if input_data.get("产品名称") == '全部':
        FE.table().select_all()
    else:
        FE.table().select_row({'产品名称': input_data.get("产品名称")}, True)
    # 表单填入
    click_delete()
    click_confirm()
    # 检查查询结果
    if input_data.get("产品名称") == '全部':
        check_table_rows(input_data=None, ui_expect=ui_expect)
    else:
        FE.table().check_cells_exist({'产品名称': input_data.get("产品名称")})
    # 检查查询结果
    check_table_rows(input_data=None, ui_expect=ui_expect)

def special_product_delete_cancel(input_data=None, ui_expect=None):
    # 进入 风险等级查询 菜单 线下客户管理/线下客户管理
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '线下客户管理/专户产品管理'})
    click_delete()

  # 资源设置搜索框表单填入
    FE.table().select_row({'产品名称': input_data.get("产品名称")}, True)

    # 表单填入
    click_delete()
    time.sleep(1)

    click_cancel()
    time.sleep(1)
    FE.table().check_cells_exist({'客户名称': ui_expect.get("客户名称")})